import pandas as pd

def create_excel_template():
    """
    创建一个用于Milvus数据导入的Excel模板文件。
    
    模板包含'general_faq'集合所需的所有非向量字段。
    """
    # 定义Excel文件的列名，这些对应于Milvus collection的字段名
    # 注意：我们在这里不包括向量字段（如 question_embedding），
    # 因为它们将由导入脚本自动生成。
    columns = [
        "faq_id",
        "question",
        "answer",
        "keywords",
        "category",
        "scene_tags",
        "emotion_type",
        "priority",
        "use_count",
        "satisfaction_rate",
        "update_time"
    ]
    
    # 创建一个空的DataFrame，并设置列名
    df = pd.DataFrame(columns=columns)
    
    # 定义示例数据，帮助用户理解格式
    sample_data = {
        "faq_id": [1, 2],
        "question": ["如何办理信用卡？", "忘记了储蓄卡密码怎么办？"],
        "answer": ["您可以通过我们的手机银行App、官方网站或前往任意网点柜台申请办理信用卡。", "请您本人携带有效身份证件及银行卡，前往我行任意网点办理密码重置业务。"],
        "keywords": ["信用卡,申请,办理", "忘记密码,储蓄卡,密码重置"], # 关键词请用英文逗号分隔
        "category": ["信用卡业务", "借记卡业务"],
        "scene_tags": ["新用户,在线申请", "密码问题,线下办理"], # 标签也请用英文逗号分隔
        "emotion_type": ["中性", "焦虑"],
        "priority": [1, 2],
        "use_count": [150, 230],
        "satisfaction_rate": [0.95, 0.92],
        "update_time": [1678886400, 1678887000] # 请使用Unix时间戳
    }
    
    df = pd.DataFrame(sample_data)

    # 将DataFrame保存到Excel文件
    output_filename = "import_data_template.xlsx"
    df.to_excel(output_filename, index=False)
    
    print(f"成功创建Excel模板文件：'{output_filename}'")
    print("请打开此文件，并按照示例格式填写您的数据。")

if __name__ == "__main__":
    create_excel_template() 